home *** CD-ROM | disk | FTP | other *** search
/ CU Amiga Super CD-ROM 15 / CU Amiga Magazine's Super CD-ROM 15 (1997)(EMAP Images)(GB)[!][issue 1997-10].iso / CUCD / Graphics / Ghostscript / source / gdevjpeg.c < prev    next >
C/C++ Source or Header  |  1997-03-23  |  8KB  |  289 lines
  1. /* Copyright (C) 1997 Aladdin Enterprises.  All rights reserved.
  2.   
  3.   This file is part of Aladdin Ghostscript.
  4.   
  5.   Aladdin Ghostscript is distributed with NO WARRANTY OF ANY KIND.  No author
  6.   or distributor accepts any responsibility for the consequences of using it,
  7.   or for whether it serves any particular purpose or works at all, unless he
  8.   or she says so in writing.  Refer to the Aladdin Ghostscript Free Public
  9.   License (the "License") for full details.
  10.   
  11.   Every copy of Aladdin Ghostscript must include a copy of the License,
  12.   normally in a plain ASCII text file named PUBLIC.  The License grants you
  13.   the right to copy, modify and redistribute Aladdin Ghostscript, but only
  14.   under certain conditions described in the License.  Among other things, the
  15.   License requires that the copyright notice and this notice be preserved on
  16.   all copies.
  17. */
  18.  
  19. /* gdevjpeg.c */
  20. /* JPEG output driver */
  21. #include "stdio_.h"        /* for jpeglib.h */
  22. #include "jpeglib.h"
  23. #include "gdevprn.h"
  24. #include "stream.h"
  25. #include "strimpl.h"
  26. #include "sdct.h"
  27. #include "sjpeg.h"
  28.  
  29. /* Structure for the JPEG-writing device. */
  30. typedef struct gx_device_jpeg_s {
  31.     gx_device_common;
  32.     gx_prn_device_common;
  33.     /* Additional parameters */
  34.     int JPEGQ;        /* quality on IJG scale */
  35.     float QFactor;        /* quality per DCTEncode conventions */
  36.     /* JPEGQ overrides QFactor if both are specified. */
  37. } gx_device_jpeg;
  38.  
  39. /* The device descriptor */
  40. private dev_proc_get_params(jpeg_get_params);
  41. private dev_proc_put_params(jpeg_put_params);
  42. private dev_proc_print_page(jpeg_print_page);
  43.  
  44. /* ------ The device descriptors ------ */
  45.  
  46. /* Default X and Y resolution. */
  47. #ifndef X_DPI
  48. #  define X_DPI 72
  49. #endif
  50. #ifndef Y_DPI
  51. #  define Y_DPI 72
  52. #endif
  53.  
  54. /* 24-bit color */
  55.  
  56. private const gx_device_procs jpeg_procs =
  57.   prn_color_params_procs(gdev_prn_open, gdev_prn_output_page, gdev_prn_close,
  58.              gx_default_rgb_map_rgb_color,
  59.              gx_default_rgb_map_color_rgb,
  60.              jpeg_get_params, jpeg_put_params);
  61.  
  62. gx_device_jpeg far_data gs_jpeg_device =
  63. { prn_device_std_body(gx_device_jpeg, jpeg_procs, "jpeg",
  64.               DEFAULT_WIDTH_10THS, DEFAULT_HEIGHT_10THS,
  65.               X_DPI, Y_DPI, 0,0,0,0, 24, jpeg_print_page),
  66.   0,    /* JPEGQ: 0 indicates not specified */
  67.   0.0    /* QFactor: 0 indicates not specified */
  68. };
  69.  
  70. /* 8-bit gray */
  71.  
  72. private const gx_device_procs jpeggray_procs =
  73.   prn_color_params_procs(gdev_prn_open, gdev_prn_output_page, gdev_prn_close,
  74.              gx_default_gray_map_rgb_color,
  75.              gx_default_gray_map_color_rgb,
  76.              jpeg_get_params, jpeg_put_params);
  77.  
  78. gx_device_jpeg far_data gs_jpeggray_device =
  79. { prn_device_body(gx_device_jpeg, jpeggray_procs, "jpeggray",
  80.           DEFAULT_WIDTH_10THS, DEFAULT_HEIGHT_10THS,
  81.           X_DPI, Y_DPI, 0,0,0,0,
  82.           1,8,255,0,256,0,
  83.           jpeg_print_page),
  84.   0,    /* JPEGQ: 0 indicates not specified */
  85.   0.0    /* QFactor: 0 indicates not specified */
  86. };
  87.  
  88. /* Get parameters. */
  89. private int
  90. jpeg_get_params(gx_device *dev, gs_param_list *plist)
  91. {    gx_device_jpeg *jdev = (gx_device_jpeg *)dev;
  92.     int code = gdev_prn_get_params(dev, plist);
  93.     int ecode;
  94.  
  95.     if ( code < 0 )
  96.       return code;
  97.  
  98.     if ((ecode = param_write_int(plist, "JPEGQ", &jdev->JPEGQ)) < 0)
  99.       code = ecode;
  100.     if ((ecode = param_write_float(plist, "QFactor", &jdev->QFactor)) < 0)
  101.       code = ecode;
  102.  
  103.     return code;
  104. }
  105.  
  106. /* Put parameters. */
  107. private int
  108. jpeg_put_params(gx_device *dev, gs_param_list *plist)
  109. {    gx_device_jpeg *jdev = (gx_device_jpeg *)dev;
  110.     int ecode = 0;
  111.     int code;
  112.     gs_param_name param_name;
  113.     int jq = jdev->JPEGQ;
  114.     float qf = jdev->QFactor;
  115.  
  116.     switch ( code = param_read_int(plist, (param_name = "JPEGQ"), &jq) )
  117.     {
  118.     case 0:
  119.         if ( jq < 0 || jq > 100 )
  120.           ecode = gs_error_limitcheck;
  121.         else
  122.           break;
  123.         goto jqe;
  124.     default:
  125.         ecode = code;
  126. jqe:        param_signal_error(plist, param_name, ecode);
  127.     case 1:
  128.         break;
  129.     }
  130.  
  131.     switch ( code = param_read_float(plist, (param_name = "QFactor"), &qf) )
  132.     {
  133.     case 0:
  134.         if ( qf < 0.0 || qf > 1.0e6 )
  135.           ecode = gs_error_limitcheck;
  136.         else
  137.           break;
  138.         goto qfe;
  139.     default:
  140.         ecode = code;
  141. qfe:        param_signal_error(plist, param_name, ecode);
  142.     case 1:
  143.         break;
  144.     }
  145.  
  146.     if ( ecode < 0 )
  147.       return ecode;
  148.     code = gdev_prn_put_params(dev, plist);
  149.     if ( code < 0 )
  150.       return code;
  151.  
  152.     jdev->JPEGQ = jq;
  153.     jdev->QFactor = qf;
  154.     return 0;
  155. }
  156.  
  157. /* Send the page to the file. */
  158. private int
  159. jpeg_print_page(gx_device_printer *pdev, FILE *prn_stream)
  160. {    gx_device_jpeg *jdev = (gx_device_jpeg *)pdev;
  161.     gs_memory_t *mem = pdev->memory;
  162.     int line_size = gdev_mem_bytes_per_scan_line((gx_device *)pdev);
  163.     byte *in = gs_alloc_bytes(mem, line_size, "jpeg_print_page(in)");
  164.     /* The current implementation of the DCTE filter */
  165.     /* requires that we allocate this with gs_malloc! */
  166.     jpeg_compress_data *jcdp =
  167.       gs_malloc(1, sizeof(*jcdp), "jpeg_print_page(jpeg_compress_data)");
  168.     byte *fbuf = 0;
  169.     uint fbuf_size;
  170.     byte *jbuf = 0;
  171.     uint jbuf_size;
  172.     int lnum;
  173.     int code;
  174.     stream_DCT_state state;
  175.     stream fstrm, jstrm;
  176.  
  177.     static const stream_procs filter_write_procs =
  178.     { s_std_noavailable, s_std_noseek, s_std_write_reset,
  179.       s_std_write_flush, s_filter_close
  180.     };
  181.  
  182.     if ( jcdp == 0 || in == 0 )
  183.       { code = gs_note_error(gs_error_VMerror);
  184.         goto fail;
  185.       }
  186.     /* Create the DCT decoder state. */
  187.     state.template = &jcdp->template;
  188.     state.memory = 0;
  189.     state.QFactor = 1.0;    /* disable quality adjustment in zfdcte.c */
  190.     state.ColorTransform = 1; /* default for RGB */
  191.     /* We insert no markers, allowing the IJG library to emit */
  192.     /* the format it thinks best. */
  193.     state.NoMarker = true;    /* do not insert our own Adobe marker */
  194.     state.Markers.data = 0;
  195.     state.Markers.size = 0;
  196.     state.data.compress = jcdp;
  197.     if ( (code = gs_jpeg_create_compress(&state)) < 0 )
  198.       goto fail;
  199.     jcdp->cinfo.image_width = pdev->width;
  200.     jcdp->cinfo.image_height = pdev->height;
  201.     switch (pdev->color_info.depth) {
  202.       case 24:
  203.         jcdp->cinfo.input_components = 3;
  204.         jcdp->cinfo.in_color_space = JCS_RGB;
  205.         break;
  206.       case 8:
  207.         jcdp->cinfo.input_components = 1;
  208.         jcdp->cinfo.in_color_space = JCS_GRAYSCALE;
  209.         break;
  210.     }
  211.     /* Set compression parameters. */
  212.     if ( (code = gs_jpeg_set_defaults(&state)) < 0 )
  213.       goto done;
  214.     if (jdev->JPEGQ > 0)
  215.       { code = gs_jpeg_set_quality(&state, jdev->JPEGQ, TRUE);
  216.         if ( code < 0 )
  217.           goto done;
  218.       }
  219.     else if (jdev->QFactor > 0.0)
  220.       { code = gs_jpeg_set_linear_quality(&state,
  221.                           (int) (min(jdev->QFactor, 100.0)
  222.                              * 100.0 + 0.5),
  223.                           TRUE);
  224.         if ( code < 0 )
  225.           goto done;
  226.       }
  227.     jcdp->cinfo.restart_interval = 0;
  228.     jcdp->cinfo.density_unit = 1; /* dots/inch (no #define or enum) */
  229.     jcdp->cinfo.X_density = pdev->HWResolution[0];
  230.     jcdp->cinfo.Y_density = pdev->HWResolution[1];
  231.     /* Create the filter. */
  232.     jcdp->template = s_DCTE_template;
  233.     /* Make sure we get at least a full scan line of input. */
  234.     state.scan_line_size = jcdp->cinfo.input_components *
  235.                    jcdp->cinfo.image_width;
  236.     jcdp->template.min_in_size =
  237.       max(s_DCTE_template.min_in_size, state.scan_line_size);
  238.     /* Make sure we can write the user markers in a single go. */
  239.     jcdp->template.min_out_size =
  240.       max(s_DCTE_template.min_out_size, state.Markers.size);
  241.  
  242.     /* Set up the streams. */
  243.     fbuf_size = max(512 /* arbitrary */, jcdp->template.min_out_size);
  244.     jbuf_size = jcdp->template.min_in_size;
  245.     if ( (fbuf = gs_alloc_bytes(mem, fbuf_size, "jpeg_print_page(fbuf)")) == 0 ||
  246.          (jbuf = gs_alloc_bytes(mem, jbuf_size, "jpeg_print_page(jbuf)")) == 0
  247.        )
  248.       { code = gs_note_error(gs_error_VMerror);
  249.         goto done;
  250.       }
  251.     swrite_file(&fstrm, prn_stream, fbuf, fbuf_size);
  252.     s_std_init(&jstrm, jbuf, jbuf_size, &filter_write_procs,
  253.            s_mode_write);
  254.     jstrm.memory = mem;
  255.     jstrm.state = (stream_state *)&state;
  256.     jstrm.procs.process = state.template->process;
  257.     jstrm.strm = &fstrm;
  258.     if ( state.template->init )
  259.       (*state.template->init)(jstrm.state);
  260.  
  261.     /* Copy the data to the output. */
  262.     for ( lnum = 0; lnum < pdev->height; ++lnum )
  263.       { byte *data;
  264.         uint ignore_used;
  265.  
  266.         gdev_prn_get_bits(pdev, lnum, in, &data);
  267.         sputs(&jstrm, data, state.scan_line_size, &ignore_used);
  268.       }
  269.  
  270.     /* Wrap up. */
  271.     sclose(&jstrm);
  272.     sflush(&fstrm);
  273.     jcdp = 0;
  274. done:
  275.     gs_free_object(mem, jbuf, "jpeg_print_page(jbuf)");
  276.     gs_free_object(mem, fbuf, "jpeg_print_page(fbuf)");
  277.     if ( jcdp )
  278.       gs_jpeg_destroy(&state); /* frees *jcdp */
  279.     gs_free_object(mem, in, "jpeg_print_page(in)");
  280.     return code;
  281. fail:
  282.     if ( jcdp )
  283.       gs_free(jcdp, 1, sizeof(*jcdp),
  284.           "jpeg_print_page(jpeg_compress_data)");
  285.     gs_free_object(mem, in, "jpeg_print_page(in)");
  286.     return code;
  287. #undef jcdp
  288. }
  289.